from docx import Document
from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
from docx.shared import Pt
from docx.oxml.ns import qn
import os
import time

# 获取用户输入
price = input("请输入调整的工资：")
employee_list = ["员工1", "员工2", "员工3"]
today = time.strftime("%Y年%m月%d日", time.localtime())

# 获取脚本所在目录
script_dir = os.path.dirname(os.path.abspath(__file__))
os.makedirs(script_dir, exist_ok=True)  # 确保目录存在

for i in employee_list:
    # 1️⃣ 创建 Word 文档
    doc = Document()

    # 2️⃣ 设置默认样式（仅设置中文字体）
    style = doc.styles['Normal']
    style.font.name = '微软雅黑'  # 拉丁字体
    style._element.rPr.rFonts.set(qn('w:eastAsia'), '微软雅黑')  # 中文字体
    style.font.size = Pt(12)

    # 3️⃣ 添加标题并设置样式
    paragraph = doc.add_paragraph()
    #居中对齐
    paragraph.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
    run = paragraph.add_run("工资调整通知")
    run.bold = True  # 加粗
    run.font.size = Pt(21)  # 字号
    run.space_after = Pt(12)  # 段后间距
    run.space_before = Pt(12)  # 段前间距
    run.font.name = '微软雅黑'  # 拉丁字体
    run._element.rPr.rFonts.set(qn('w:eastAsia'), '微软雅黑')  # 中文字体

    # 添加正文段落 - 敬尊的 [姓名]：
    p = doc.add_paragraph()
    run1= p.add_run("尊敬的 ")  # 普通文字
    run1.font.size = Pt(16)  # 字号
    bold_run = p.add_run(i)  # 员工名
    bold_run.bold = True  # 加粗显示
    bold_run.font.size = Pt(16)  # 字号
    p.add_run("：")  # 结尾冒号


    p2=doc.add_paragraph()
    run2= p2.add_run(f"根据公司政策，您的工资将调整为：{price}元")
    run2.font.size = Pt(14)  # 字号

    p3=doc.add_paragraph()
    p3.alignment = WD_PARAGRAPH_ALIGNMENT.RIGHT  # 右对齐
    run4=p3.add_run("特此通知")
    run4.font.size = Pt(14)  # 字号
    run4.font.bold = True  # 加粗

    # 5️⃣ 保存文件
    file_name = os.path.join(script_dir, f"{i}_工资调整通知.docx")
    doc.save(file_name)
    print(f"✅ 已生成文件：{file_name}")